#include "../mq_server/mq_exchange.hpp"
#include <gtest/gtest.h>

using namespace mq_server;

ExchangeManager::ptr emp;

class ExchangeTest : public testing::Environment {
    public:
        virtual void SetUp() override {
            emp = std::make_shared<ExchangeManager>("./data/meta.db");
        }
        virtual void TearDown() override {
            emp->clear();
            std::cout << "最后的清理！！\n";
        }
};

TEST(exchange_test, insert_test) {
    std::unordered_map<std::string, std::string> map = {{"k1", "v1"},{"k2", "v2"}};
    emp->DeclareExchange("exchange1", ExchangeType::DIRECT, true, false, map);
    emp->DeclareExchange("exchange2", ExchangeType::DIRECT, true, false, map);
    emp->DeclareExchange("exchange3", ExchangeType::DIRECT, true, false, map);
    emp->DeclareExchange("exchange4", ExchangeType::DIRECT, true, false, map);
    ASSERT_EQ(emp->size(), 4);
}
TEST(exchange_test, select_test) {
    ASSERT_EQ(emp->Exists("exchange1"), true);
    ASSERT_EQ(emp->Exists("exchange2"), true);
    ASSERT_EQ(emp->Exists("exchange3"), true);
    ASSERT_EQ(emp->Exists("exchange4"), true);
    Exchange::ptr exp = emp->SelectExchange("exchange2");
    ASSERT_NE(exp.get(), nullptr);
    ASSERT_EQ(exp->_name, "exchange2");
    ASSERT_EQ(exp->_durable, true);
    ASSERT_EQ(exp->_auto_delete, false);
    ASSERT_EQ(exp->_type, ExchangeType::DIRECT);
    ASSERT_EQ(exp->getArgs(), std::string("k2=v2&k1=v1"));
}

TEST(exchange_test, remove_test) {
    emp->DeleteExchange("exchange2");
    Exchange::ptr exp = emp->SelectExchange("exchange2");
    ASSERT_EQ(exp.get(), nullptr);
    ASSERT_EQ(emp->Exists("exchange2"), false);
}

int main(int argc,char *argv[])
{
    testing::InitGoogleTest(&argc, argv);
    testing::AddGlobalTestEnvironment(new ExchangeTest);
    RUN_ALL_TESTS();
    return 0;
}